package com.springboot.fram.core.lang;

import org.apache.commons.lang3.math.NumberUtils;

import java.math.BigDecimal;
import java.math.RoundingMode;

import static org.apache.commons.lang3.StringUtils.*;

/**
 * 浮点数精确计算
 * QC
 * 2016年12月21日 下午2:15:19
 */
public class ArithmeticUtils {
    private static final String[] MONEY_UPPERS = new String[]{
            "零", "壹", "贰", "叁", "肆", "伍", "陆", "柒", "捌", "玖"
    };
    private static final String[] MONEY_UNITS = new String[]{
            "圆", "拾", "佰", "仟", "万", "拾", "佰", "仟", "亿", "拾", "佰", "仟", "万"
    };
    private static final String[] SMALL_UNITS = new String[]{"分", "角"};
    private static final String[] tuaRegexp1 = new String[]{"零仟", "零佰", "零拾"};
    private static final String[] tuaRegexp2 = new String[]{"零亿", "零万", "零圆"};
    private static final String[] tuaRegexp3 = new String[]{"亿", "万", "圆"};
    private static final String[] tuaRegexp4 = new String[]{"零角", "零分"};

    /**
     * 将给定的表示金钱的数转换成中文大写形式。
     * <pre>
     * StringUtils.toUpperAmount(BigDecimal.valueOf(1.05)) = "壹圆伍分"
     * StringUtils.toUpperAmount(BigDecimal.valueOf(40000001.1)) = "肆仟万零壹圆壹角"
     * StringUtils.toUpperAmount(BigDecimal.valueOf(123456.78)) = "壹拾贰万叁仟肆佰伍拾陆圆柒角捌分"
     * </pre>
     *
     * @param amount 表示金钱的数。
     * @return 给定的表示金钱的数的中文大写形式。
     */
    public static String toUpperAmount(BigDecimal amount) {
        if (amount == null) {
            return EMPTY;
        }
        String strAmount = NumberUtils.toScaledBigDecimal(amount, NumberUtils.INTEGER_TWO, RoundingMode.UP).toString();
        String[] strAmounts = split(strAmount, '.');
        int[] his = _toUpperAmountToInts(strAmounts[0]), lis = _toUpperAmountToInts(strAmounts[1]);
        int hisl = his.length, lisl = lis.length;

        StringBuffer buf = new StringBuffer();
        for (int i = 0; i < hisl; i++) {
            buf.append(MONEY_UPPERS[his[i]]).append(MONEY_UNITS[hisl - i - 1]);
        }
        for (int i = 0; i < lisl; i++) {
            buf.append(MONEY_UPPERS[lis[i]]).append(SMALL_UNITS[lisl - i - 1]);
        }
        if (buf.length() == 0) {
            return MONEY_UPPERS[0];
        }
        while (buf.charAt(0) == '零') {
            buf.delete(0, 2);
        }
        String result = buf.toString();
        for (String s : tuaRegexp1) {
            result = replace(result, s, MONEY_UPPERS[0]);
        }
        final String repeatZero1 = "零零零", repeatZero2 = "零零";
        for (int i = 0; i < 3; i++) {
            result = replace(result, repeatZero1, MONEY_UPPERS[0]);
            result = replace(result, repeatZero2, MONEY_UPPERS[0]);
            result = replace(result, tuaRegexp2[i], tuaRegexp3[i]);
        }
        for (int i = 0; i < 2; i++) {
            result = replace(result, tuaRegexp4[i], EMPTY);
        }
        return result;
    }

    private static int[] _toUpperAmountToInts(String amount) {
        int[] ints = new int[amount.length()];
        for (int i = 0, l = amount.length(); i < l; i++) {
            ints[i] = Integer.parseInt(String.valueOf(amount.charAt(i)));
        }
        return ints;
    }


    /**
     * 除法精度（除不尽时保留10位小数）
     */
    private static final int DIV_SCALE = 10;

    /**
     * 元转分，确保price保留两位有效数字
     *
     * @return
     */
    public static long yuanToFen(String amount) {
//        BigDecimal bigDecimal = new BigDecimal(amount).setScale(2, BigDecimal.ROUND_HALF_UP);
//        return String.valueOf(bigDecimal.multiply(new BigDecimal(100)).longValue());
        return new BigDecimal(String.valueOf(amount)).movePointRight(2).longValue();
    }

    /**
     * 分转元，转换为bigDecimal在toString
     *
     * @return
     */
    public static BigDecimal fenToYuan(long amount) {
        // 小数点向左移动位数
        return new BigDecimal(Long.valueOf(amount).toString()).movePointLeft(2);
        // return BigDecimal.valueOf(price).divide(new BigDecimal(100)).toString();
    }


    /**
     * 小数精确加法
     *
     * @param d1 参数1
     * @param d2 参数2
     * @return double
     */
    public static double add(double d1, double d2) {
        BigDecimal bd1 = BigDecimal.valueOf(d1);
        BigDecimal bd2 = BigDecimal.valueOf(d2);
        return bd1.add(bd2).doubleValue();
    }

    /**
     * 提供精确的加法运算
     *
     * @param v1 被加数
     * @param v2 加数
     * @return 两个参数的和
     */
    public static BigDecimal add(String v1, String v2) {
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return b1.add(b2);
    }

    /**
     * 提供精确的加法运算
     *
     * @param v1    被加数
     * @param v2    加数
     * @param scale 保留scale 位小数
     * @return 两个参数的和
     */
    public static String add(String v1, String v2, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return b1.add(b2).setScale(scale, BigDecimal.ROUND_HALF_UP).toString();
    }


    /**
     * 小数精确减法
     *
     * @param d1 参数1
     * @param d2 参数2
     * @return double
     */
    public static double sub(double d1, double d2) {
        BigDecimal bd1 = BigDecimal.valueOf(d1);
        BigDecimal bd2 = BigDecimal.valueOf(d2);
        return bd1.subtract(bd2).doubleValue();
    }

    /**
     * 提供精确的减法运算。
     *
     * @param v1 被减数
     * @param v2 减数
     * @return 两个参数的差
     */
    public static BigDecimal sub(String v1, String v2) {
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return b1.subtract(b2);
    }

    /**
     * 提供精确的减法运算
     *
     * @param v1    被减数
     * @param v2    减数
     * @param scale 保留scale 位小数
     * @return 两个参数的差
     */
    public static String sub(String v1, String v2, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return b1.subtract(b2).setScale(scale, BigDecimal.ROUND_HALF_UP).toString();
    }

    /**
     * 提供精确的乘法运算
     *
     * @param v1 被乘数
     * @param v2 乘数
     * @return 两个参数的积
     */
    public static BigDecimal mul(String v1, String v2) {
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return b1.multiply(b2);
    }

    /**
     * 小数精确乘法
     *
     * @param d1 参数1
     * @param d2 参数2
     * @return double
     */
    public static double mul(double d1, double d2) {
        BigDecimal bd1 = BigDecimal.valueOf(d1);
        BigDecimal bd2 = BigDecimal.valueOf(d2);
        return bd1.multiply(bd2).doubleValue();
    }

    /**
     * 小数精确乘法
     *
     * @param d1    参数1
     * @param d2    参数2
     * @param scale 保留小数位
     * @return double
     */
    public static double mul(double d1, double d2, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal bd1 = BigDecimal.valueOf(d1);
        BigDecimal bd2 = BigDecimal.valueOf(d2);
        return bd1.multiply(bd2).setScale(scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 提供精确的乘法运算
     *
     * @param v1    被乘数
     * @param v2    乘数
     * @param scale 保留scale 位小数
     * @return 两个参数的积
     */
    public static String mul(String v1, String v2, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return b1.multiply(b2).setScale(scale, BigDecimal.ROUND_HALF_UP).toString();
    }

    /**
     * 小数精确除法，默认保留10位小数
     *
     * @param d1 参数1
     * @param d2 参数2
     * @return double
     */
    public static double div(double d1, double d2) {
        BigDecimal bd1 = BigDecimal.valueOf(d1);
        BigDecimal bd2 = BigDecimal.valueOf(d2);
        /*
         * 当除不尽时，以四舍五入的方式
         */
        return bd1.divide(bd2, DIV_SCALE, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 小数精确除法保留scale位小数
     *
     * @param d1    参数1
     * @param d2    参数2
     * @param scale 保留小数位
     * @return double
     */
    public static double div(double d1, double d2, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal bd1 = BigDecimal.valueOf(d1);
        BigDecimal bd2 = BigDecimal.valueOf(d2);
        /*
         * 当除不尽时，以四舍五入的方式
         */
        return bd1.divide(bd2, scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 提供（相对）精确的除法运算。当发生除不尽的情况时，由scale参数指
     * 定精度，以后的数字四舍五入
     *
     * @param v1    被除数
     * @param v2    除数
     * @param scale 表示需要精确到小数点以后几位
     * @return 两个参数的商
     */
    public static String div(String v1, String v2, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return b1.divide(b2, scale, BigDecimal.ROUND_HALF_UP).toString();
    }

    /**
     * 将浮点数保留scale位小数
     *
     * @param number 数
     * @param scale  小数位
     * @return double
     */
    public static double setScale(double number, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        return BigDecimal.valueOf(number).setScale(scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * @param data
     * @return double  返回类型
     * @throws
     * @Title setScale
     * @Description TODO(保留一位小数)
     */
    public static double setScale(double data) {
        return BigDecimal.valueOf(data).setScale(1, BigDecimal.ROUND_HALF_DOWN).doubleValue();
    }

    /**
     * 提供精确的小数位四舍五入处理
     *
     * @param v     需要四舍五入的数字
     * @param scale 小数点后保留几位
     * @return 四舍五入后的结果
     */
    public static double round(double v, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal b = new BigDecimal(Double.toString(v));
        return b.setScale(scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 提供精确的小数位四舍五入处理
     *
     * @param v     需要四舍五入的数字
     * @param scale 小数点后保留几位
     * @return 四舍五入后的结果
     */
    public static String round(String v, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        BigDecimal b = new BigDecimal(v);
        return b.setScale(scale, BigDecimal.ROUND_HALF_UP).toString();
    }

    /**
     * 取余数
     *
     * @param v1    被除数
     * @param v2    除数
     * @param scale 小数点后保留几位
     * @return 余数
     */
    public static String remainder(String v1, String v2, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return b1.remainder(b2).setScale(scale, BigDecimal.ROUND_HALF_UP).toString();
    }

    /**
     * 取余数  BigDecimal
     *
     * @param v1    被除数
     * @param v2    除数
     * @param scale 小数点后保留几位
     * @return 余数
     */
    public static BigDecimal remainder(BigDecimal v1, BigDecimal v2, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        return v1.remainder(v2).setScale(scale, BigDecimal.ROUND_HALF_UP);
    }

    /**
     * 比较大小
     *
     * @param v1 被比较数
     * @param v2 比较数
     * @return 如果v1 大于v2 则 返回true 否则false
     */
    public static boolean compare(String v1, String v2) {
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        int bj = b1.compareTo(b2);
        if (bj > 0) {
            return true;
        }
        return false;
    }
}
